import cv2

# 定义 RTSP 流地址
rtsp_url = "rtsp://192.168.16.181:8554/live"

# 使用 OpenCV 连接到 RTSP 流
cap = cv2.VideoCapture(rtsp_url)

# 检查是否成功连接
if not cap.isOpened():
    print("无法连接到 RTSP 流")
else:
    print("成功连接到 RTSP 流")

# 开始读取图像数据并显示
while cap.isOpened():
    # 读取当前帧
    ret, frame = cap.read()

    # 检查是否成功读取帧
    if ret:
        # 显示帧
        cv2.imshow("无人机图像数据", frame)
        
        # 按 'q' 键退出
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        print("无法读取图像数据")
        break

# 释放视频捕获对象并关闭所有窗口
cap.release()
cv2.destroyAllWindows()
